import torch

x_cpu = torch.tensor([1, 1, float('nan'), float('nan')], dtype=torch.float32)
unique_cpu, counts_cpu = torch.unique(x_cpu, return_counts=True)
print("CPU Result:", unique_cpu, counts_cpu)

if torch.cuda.is_available():
    x_gpu = x_cpu.cuda()
    unique_gpu, counts_gpu = torch.unique(x_gpu, return_counts=True)
    print("GPU Result:", unique_gpu.cpu(), counts_gpu.cpu())
else:
    print("GPU not available")